tensor([[0.6600, 0.6600, 0.6600, 0.6600, 0.6600],
        [0.3400, 0.3400, 0.3400, 0.3400, 0.3400]], grad_fn=<RoundBackward1>)